import numpy as np

if __name__ == '__main__':
    matrix_1d = np.array([1, 2, 3, 4])
    print(matrix_1d)

    matrix_2d = np.array(
        [
            [1, 2, 3],
            [4, 5, 6]
        ]
    )
    print(matrix_2d)

    matrix_3d = np.array(
        [
            [
                [1, 2],
                [3, 4]
            ],
            [
                [5, 6],
                [7, 8]
            ],
            [
                [9, 0],
                [1, 5]
            ]
        ]
    )

    print(matrix_3d)
